from rest_framework.pagination import PageNumberPagination
from rest_framework.response import Response


def jwt_response_payload_handler(token, user=None, request=None):
    return {
        'token': token,
        'username': user.username,
        'id': user.id
    }


# 自定义分页类
class PageNum(PageNumberPagination):
    # 后端指定显示数量
    page_size = 5
    page_size_query_param = 'pagesize'
    max_page_size = 10

    # 重写分页返回方法，按照指定的字段进行分页数据返回
    def get_paginated_response(self, data):
        return Response({
            # 总数量
            'count': self.page.paginator.count,
            # 用户数据
            'lists': data,
            # 当前页数
            'page': self.page.number,
            # 总页数
            'pages': self.page.paginator.num_pages,
            # 后端指定的页容量
            'pagesize': self.page_size,
        })
